How to check diagonal of a square